# -*- coding: utf-8 -*-
#当前所属项目：ailearning
#文件名：prompt
#当前用户：李峰
#当前时间：2025/9/20 22:54
#模块说明：提示词
from langchain_openai import ChatOpenAI

def prompt():
    "使用提示词对与大模型交互"
    llm = ChatOpenAI(model_name='qwen-plus', base_url="https://dashscope.aliyuncs.com/compatible-mode/v1", temperature=0.5,
                     api_key="sk-7b4f290b1a3e42a1a9a1957fa44eff37")
    prompt="今天是星期一，明天是星期三"
    re = llm.invoke(prompt)
    print(re.content)

def prompt_mb():
    """
    提示词模版
    想查看一些提示词模版可以访问这个网站
    https://smith.langchain.com/hub
    :return:
    """
    from langchain_core.prompts import PromptTemplate
    llm = ChatOpenAI(model_name='qwen-plus', base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
                     temperature=0.5,
                     api_key="sk-7b4f290b1a3e42a1a9a1957fa44eff37")
    template="请回答：{location}"
    prompt =PromptTemplate.from_template(template)
    final_prompt = prompt.format(location="耒阳是哪个省的？")
    print(f"最终提示词:{final_prompt}")
    print("__________________")
    print(f"LLM输出：{llm.invoke(final_prompt)}")



if __name__ == "__main__":
    # prompt()
    prompt_mb()